import PyPDF2
# 合并指定PDF
def merge_pdfs(input_pdf_paths, output_pdf_path):
    # 创建一个 PdfWriter 对象，用于将多个 PDF 页面合并到一个新的 PDF 文件中
    pdf_writer = PyPDF2.PdfWriter()

    # 遍历输入的 PDF 文件路径列表
    for pdf_path in input_pdf_paths:
        # 以二进制只读模式打开当前的 PDF 文件
        with open(pdf_path, 'rb') as file:
            # 创建一个 PdfReader 对象，用于读取当前 PDF 文件的内容
            pdf_reader = PyPDF2.PdfReader(file)
            # 遍历当前 PDF 文件的每一页
            for page_num in range(len(pdf_reader.pages)):
                # 获取当前页
                page = pdf_reader.pages[page_num]
                # 将当前页添加到 PdfWriter 对象中
                pdf_writer.add_page(page)

    # 以二进制写入模式打开输出的 PDF 文件
    with open(output_pdf_path, 'wb') as output_file:
        # 将 PdfWriter 对象中的所有页面写入到输出的 PDF 文件中
        pdf_writer.write(output_file)

if __name__ == "__main__":
    # 输入的两个 PDF 文件的路径，你可以根据实际情况修改
    input_pdf_paths = ['kcshow/step11.pdf', 'kcshow/step12.pdf']
    # 输出合并后 PDF 文件的路径，你可以根据实际情况修改
    output_pdf_path = 'kcshow/step11-12.pdf'
    # 调用合并函数
    merge_pdfs(input_pdf_paths, output_pdf_path)